﻿using System;
using System.Runtime.InteropServices;

namespace StockUtil
{
    public class StructHelper
    {
        /// <summary>
        /// 结构体转byte数组
        /// </summary>
        /// <param name="structObj">要转换的结构体</param>
        /// <returns>转换后的byte数组</returns>
        public static void ToBytes(object structObj, byte[] bytes, int index)
        {
            //得到结构体的大小
            int size = Marshal.SizeOf(structObj);

            //分配结构体大小的内存空间
            IntPtr structPtr = Marshal.AllocHGlobal(size);

            //将结构体拷到分配好的内存空间
            Marshal.StructureToPtr(structObj, structPtr, false);

            //从内存空间拷到byte数组
            Marshal.Copy(structPtr, bytes, index, size);

            //释放内存空间
            Marshal.FreeHGlobal(structPtr);
        }

        /// <summary>
        /// 结构体转byte数组
        /// </summary>
        /// <param name="structObj">要转换的结构体</param>
        /// <returns>转换后的byte数组</returns>
        public static byte[] ToBytes(object structObj)
        {
            //得到结构体的大小
            int size = Marshal.SizeOf(structObj);

            //创建byte数组
            byte[] bytes = new byte[size];

            //分配结构体大小的内存空间
            IntPtr structPtr = Marshal.AllocHGlobal(size);

            //将结构体拷到分配好的内存空间
            Marshal.StructureToPtr(structObj, structPtr, false);

            //从内存空间拷到byte数组
            Marshal.Copy(structPtr, bytes, 0, size);

            //释放内存空间
            Marshal.FreeHGlobal(structPtr);

            //返回byte数组
            return bytes;
        }

        /// <summary>
        /// byte数组转结构体
        /// </summary>
        /// <param name="bytes">byte数组</param>
        /// <param name="type">结构体类型</param>
        /// <returns>转换后的结构体</returns>
        public static object FromBytes(byte[] bytes, Type type)
        {
            return FromBytes(bytes, 0, type);
        }


        /// <summary>
        /// byte数组转结构体
        /// </summary>
        /// <param name="bytes">byte数组</param>
        /// <param name="type">结构体类型</param>
        /// <returns>转换后的结构体</returns>
        public static object FromBytes(byte[] bytes, int index, Type type)
        {
            if (index < 0 || index >= bytes.Length)
            {
                throw new IndexOutOfRangeException();
            }

            //得到结构体的大小
            int size = Marshal.SizeOf(type);
            //byte数组长度小于结构体的大小
            if (size + index > bytes.Length)
            {
                //返回空
                return null;
            }

            //分配结构体大小的内存空间
            IntPtr structPtr = Marshal.AllocHGlobal(size);

            //将byte数组拷到分配好的内存空间
            Marshal.Copy(bytes, index, structPtr, size);

            //将内存空间转换为目标结构体
            object obj = Marshal.PtrToStructure(structPtr, type);

            //释放内存空间
            Marshal.FreeHGlobal(structPtr);

            //返回结构体
            return obj;
        }
    }
}
